package com.zf.hashtab;

import java.util.Scanner;

public class HashTabDemo {
    public static void main(String[] args) {
        HashTab hashTab = new HashTab(7);

        //简单菜单
        String key = "";
        Scanner scanner = new Scanner(System.in);
        while (true){
            System.out.println("add:添加雇员");
            System.out.println("list:显示雇员");
            System.out.println("find:查找雇员");
            System.out.println("exit:退出系统");

            key = scanner.next();
            switch (key){
                case "add":
                    System.out.println("输入id：");
                    int id = scanner.nextInt();
                    System.out.println("输入名字:");
                    String name = scanner.next();
                    //创建雇员
                    Emp emp = new Emp(id, name);
                    hashTab.add(emp);
                    break;
                case "list":
                    hashTab.list();
                    break;
                case "exit":
                    scanner.close();
                    System.exit(0);
                case "find":
                    System.out.println("请输入要查找的id");
                    id = scanner.nextInt();
                    hashTab.findEmpById(id);
                    break;
                default:
                    break;
            }

        }
    }

}
//创建hashtab管理多条链表
class HashTab{
    private EmpLinkList[] empLinkListArray;
    private int size;
    public HashTab(int size) {
        //初始化
        this.size = size;
        empLinkListArray = new EmpLinkList[size];
        //不要忘记初始化每一条链表
        for (int i = 0; i < size; i++) {
            empLinkListArray[i] = new EmpLinkList();
        }
    }

    //添加雇员
    public void add(Emp emp){
        int emplinkedListNo = hashFun(emp.id);
        empLinkListArray[emplinkedListNo].add(emp);

    }

    //遍历hashTab
    public void list(){
        for (int i = 0; i < size; i++) {
            empLinkListArray[i].list(i);
        }
    }

    //根据输入的id，查找雇员
    public void findEmpById(int id){
        int empLinkListNo = hashFun(id);
        Emp emp = empLinkListArray[empLinkListNo].findEmpById(id);
        if (emp != null){
            System.out.printf("在第%d条链表中找到雇员 id=%d\n",(empLinkListNo+1),id);
        }else {
            System.out.println("在哈希表中没有找到该雇员");
        }
    }

    //编写散列函数，用取模法
    public int hashFun(int id){
        return id % size;
    }
}

class Emp{
    public int id;
    public String name;
    public Emp next;

    public Emp(int id, String name) {
        this.id = id;
        this.name = name;
    }
}

class EmpLinkList{
    private Emp head;

    public void add(Emp emp){
        //如果添加的是第一个雇员
        if (head == null){
            head = emp;
            return;
        }
        //如果不是第一个雇员，则使用一个辅助指针，帮助定位到最后
        Emp curEmp = head;
        while (true){
            if (curEmp.next == null){
                break;
            }
            curEmp = curEmp.next;
        }
        curEmp.next = emp;
    }
    //遍历链表信息
    public void list(int no){
        if (head == null){
            System.out.println("第"+ (no+1) +"链表为空");
            return;
        }
        System.out.print("第"+ (no+1) +"条链表信息为：");
        Emp curEmp = head;
        while (true){
            System.out.printf("=>id=%d name=%s\n", curEmp.id, curEmp.name);
            if (curEmp.next == null){
                break;
            }
            curEmp = curEmp.next;
        }
    }

    /**
     * 根据id查找雇员，如果查找到返回该雇员，否则返回null
     * @param id
     * @return
     */
    public Emp findEmpById(int id){
        //判断是否为空
        if (head == null){
            System.out.println("链表为空");
            return null;
        }

        //辅助指针
        Emp curEmp = head;
        while (true){
            if (curEmp.id == id){
                break;
            }
            if (curEmp.next == null){
                curEmp = null;
                break;
            }
            curEmp = curEmp.next;
        }
        return curEmp;
    }
}
